package com.tan.websocket.config;

import com.tan.websocket.handler.MainWebsocketFrame;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.handler.codec.http.websocketx.*;
import io.netty.handler.timeout.IdleState;
import io.netty.handler.timeout.IdleStateEvent;
import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

/**
 * WebSocket 消息处理类
 */
@Slf4j
@Component
@ChannelHandler.Sharable
public class WebSocketHandler extends SimpleChannelInboundHandler<WebSocketFrame> {

    @Resource
    private MainWebsocketFrame mainWebsocketFrame;

    /**
     * 处理接收到的 WebSocket 帧
     *
     * @param ctx   通道处理上下文
     * @param frame 接收到的 WebSocket 帧
     * @throws Exception 处理过程中可能抛出的异常
     */
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, WebSocketFrame frame) throws Exception {
        mainWebsocketFrame.handle(ctx, frame);
    }

    /**
     * 当通道有新的连接建立时触发
     *
     * @param ctx 通道处理上下文
     * @throws Exception 可能抛出的异常
     */
    @Override
    public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
        log.info("建立新连接: " + ctx.channel().remoteAddress());
    }

    /**
     * 当通道激活（连接成功）时触发
     *
     * @param ctx 通道处理上下文
     * @throws Exception 可能抛出的异常
     */
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        log.info("连接已激活：：: " + ctx.channel().remoteAddress());
    }

    /**
     * 当通道变为非活跃状态（连接关闭）时触发
     *
     * @param ctx 通道处理上下文
     * @throws Exception 可能抛出的异常
     */
    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        log.info("连接已关闭: " + ctx.channel().remoteAddress());
    }

    /**
     * 当用户事件被触发时处理
     * 这里主要处理空闲状态事件
     *
     * @param ctx 通道处理上下文
     * @param evt 触发的事件对象
     * @throws Exception 可能抛出的异常
     */
    @Override
    public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception {
        if (evt instanceof IdleStateEvent) {
            IdleStateEvent event = (IdleStateEvent) evt;
            if (event.state() == IdleState.READER_IDLE) {
                log.info("有一段时间没有收到任何数据。, 关闭连接");
                ctx.close();
            } else if (event.state() == IdleState.WRITER_IDLE) {
                log.info("有段时间没发送消息,发送一个ping帧{}",ctx.channel().remoteAddress());
                ctx.channel().writeAndFlush(new PingWebSocketFrame());
            }
        }
    }

    /**
     * 当通道发生异常时处理
     *
     * @param ctx   通道处理上下文
     * @param cause 异常对象
     */
    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) {
        log.error("{}",cause);
        log.error("通道异常时候: " + cause.getMessage());
        ctx.close();
    }
}